using System;
using System.CodeDom;

namespace AutoUnitTest.Common.CodeGeneration.CS
{
    public class MockGenerator : ClassGenerator
    {
        public override void CreateClass(string className, string nameSpace = null)
        {
            base.CreateClass("Mock"+className, nameSpace);
            Class.BaseTypes.Add(className);
        }

        public override void AddMethod(string methodName, 
            MethodParameter returnValue = null,
            MemberAttributes methodAccess = MemberAttributes.Assembly | MemberAttributes.Final, 
            params MethodParameter[] methodParameters)
        {

            base.AddMethod(methodName, returnValue, MemberAttributes.Override | MemberAttributes.Public, 
                methodParameters);
        }
    }
}
